package com.shudes.eval;

import java.util.*;

import com.shudes.game.*;

public class EquityInput {

	protected Board board;
	protected HoleCards hero;
	protected Set<HoleCards> villians;
	protected Set<Card> deadCards;	

	public EquityInput() {
		villians = new HashSet<HoleCards>(); 
		deadCards = new HashSet<Card>(); 
	}	
	
	public EquityInput(HoleCards hero, Board board, Set<HoleCards> villians, Set<Card> deadCards) { 
		this();
		this.hero=hero; 
		this.board=board; 
		
		if(villians!=null) 
			this.addVillians(villians);
		
		if(deadCards!=null) 
			this.addDeadCards(deadCards); 
	}
	
	public EquityInput(HoleCards hero, Board board, Set<HoleCards> villians) { 
		this(hero, board, villians, null); 
	}

	public void setHero(HoleCards hole) { 
		hero = hole; 
	}

	public HoleCards getHero() { 
		return hero; 
	}

	public void setBoard(Board b) { 
		board=b;
	}

	public Board getBoard() { 
		return board; 
	}
	
	public void addDeadCard(Card c) { 
		deadCards.add(c); 
	}
	
	public void addDeadCards(Collection<Card> cards) { 
		deadCards.addAll(cards); 
	}
	
	public Set<Card> getDeadCards() { 
		return deadCards; 
	}
	
	public void addVillian(HoleCards hc) { 
		villians.add(hc); 
	}
	
	public void addVillians(Set<HoleCards> hcs) { 
		villians.addAll(hcs); 
	}
	
	public Set<HoleCards> getVillians() { 
		return villians; 
	}

}
